package com.duoduo.数组;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.LinkedList;
import java.util.List;

/**
 * @program: algorithm
 * @description: 三数之和15
 * 给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有和为 0 且不重复的三元组。
 *
 * 注意：答案中不可以包含重复的三元组。
 *
 *  
 *
 * 示例 1：
 *
 * 输入：nums = [-1,0,1,2,-1,-4]
 * 输出：[[-1,-1,2],[-1,0,1]]
 * @author: chaidl  ！
 * @create: 2022-05-07 21:34
 */
public class 三数之和15 {
	public static void main(String[] args) {
		int [] arr={-1,0,1,2,-1,-4};
		threeSum1(arr);
	}
	public static List<List<Integer>> threeSum(int[] nums) {
		List<List<Integer>> results = new LinkedList<>();
		if(nums.length < 3) {
			//长度小于3,就不存在三数之和了
			return results;
		}
		Arrays.sort(nums);
		int n = nums.length;
		for (int i = 0; i < n; i++) {
			//如果第一个数就大于0，那就没必要再查找下去了
			if (nums[i] > 0) {
				break;
			}
			int left=i+1;
			int right=n-1;
			while (left< right) {
				//三数之和有三种情况：大于0，小于0，等于0
				int tmp=nums[i]+nums[left]+nums[right];
				//三数之和小于0，左边head的那个数太小了，要把它向右移
				//三数之和大于0，tail坐标的那个数太大了，要把它向左移
			    if (tmp>0){
					right--;
			    }else if (tmp < 0){
					left++;
			    }else {
					List<Integer> tmpResult=new LinkedList<>();
					tmpResult.add(nums[i]);
					tmpResult.add(nums[left]);
					tmpResult.add(nums[right]);
					results.add(tmpResult);
					//考虑到左指针和下一个元素相等
					while (left+1<=right && nums[left]==nums[left+1]){
						left++;
					}
				    //考虑到右指针和下一个元素相等
				    while (left+1<=right && nums[right]==nums[right-1]){
					    right--;
				    }
					right--;
					left++;
			    }
			}
			//考虑到当前元素和下一个元素相等
			while (i<n-1 && nums[i] == nums[i+1]){
				i++;
			}
        }
		return results;
	}
	/**
	 * 暴力解法
	 * @param nums
	 * @return
	 */
	public static List<List<Integer>> threeSum1(int[] nums) {
		List<List<Integer>> result=new LinkedList<>();
		if (nums.length < 3){
			return result;
		}
		Arrays.sort(nums);
		if (nums[0]>0){
			//如果第一个数都大于0，就不会相加为0
			return result;
		}
		for (int i = 0; i < nums.length; i++) {
			if (nums[i]>0){
				break;
			}
			int x=nums[i];
			if (i>0 && nums[i-1]==nums[i]){
				continue;
			}
			for (int j = i+1; j < nums.length; j++) {
				int y=nums[j];
				if (j>i+1 && nums[j] == nums[j-1]){
					continue;
				}
				for (int k = j+1; k < nums.length; k++) {
					int z=nums[k];
					if (k>j+1 && nums[k-1]==nums[k]) {
						break;
                    }
					if (x+y+z==0){
						List tmpList=new LinkedList();
						tmpList.add(x);
						tmpList.add(y);
						tmpList.add(z);
						result.add(tmpList);
					}
                }
			}
		}
		return result;
	}
}
